package com.vader.boot.utils;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.digest.DigestUtils;

import java.util.ArrayList;
import java.util.List;

/**
 * @author Vader
 * @date 2022/4/28
 **/
@Slf4j
public class ShortUrlUtils {

    //26 + 26 + 10 = 62
    public static final String[] CHARS = new String[]{"a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m",
            "n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z", "0", "1", "2", "3", "4", "5", "6", "7",
            "8", "9", "A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S",
            "T", "U", "V", "W", "X", "Y", "Z"};


    /**
     * 将一个长连接URL转化为一个32位的签名串，4×8，区分为四个短链
     * <p>
     * 步骤：
     * 1）将长链通过md5生成一个32位的签名串，分为四段，每个位八位
     * 2）对四段短链循环处理，取八个字节，并看成16进制串与0x3fffffff(30位1)进行与操作，即将30位以上的舍弃
     * 3）将这三十位分成六段，每五位的数字作为索引取得字符串，则可取得六位短链
     * 4）把四个短链与长链作对应跳转
     */
    public static List<String> getShortUrl(String url) {
        String md5Hex = DigestUtils.md5Hex(url);
        log.info("getShortUrl_md5_url:{}", md5Hex);

        List<String> urlList = new ArrayList<>(4);
        for (int i = 0; i < 4; i++) {
            String str = md5Hex.substring(i * 8, i * 8 + 8);

            long hexStr = Long.parseLong(str, 16) & 0x3fffffff;
            log.info("getShortUrl_hex_url:{},{}", i, md5Hex);

            StringBuilder outChar = new StringBuilder();
            for (int j = 0; j < 6; j++) {
                //0x0000003D就是十进制的61，即字符0-61
                //与运算得到的数值为61以内的数字
                long index = 0x0000003D & hexStr;

                outChar.append(CHARS[(int) index]);
                hexStr = hexStr >> 5;
            }
            urlList.add(outChar.toString());

            //放入redis即可完成跳转
//            redisTemplate.opsForValue().set("short_url_"+outChar,url);
        }

        return urlList;
    }
}
